package th.retrofit.lib;

/**
 * https://leetcode-cn.com/problems/remove-duplicates-from-sorted-array-ii/
 * 给你一个有序数组 nums ，请你 原地 删除重复出现的元素，使每个元素 最多出现两次 ，返回删除后数组的新长度。
 *
 * 不要使用额外的数组空间，你必须在 原地 修改输入数组 并在使用 O(1) 额外空间的条件下完成
 *
 */
public class Solution80 {
    public int removeDuplicates(int[] nums) {
        int nLen = nums.length;
        int index = -1;
        for (int i = 0, j = 1; j < nLen; j++) {
            if (nums[i] != nums[j]) {
                i = j;
            } else {
                if (index == -1 && (j - i) > 1) {
                    index = j - 1;
                }
            }
        }
        if (index == -1) return nLen;
        for (int i = index - 1, j = index + 1; j < nLen; j++) {
            if (nums[j] != nums[i]) {
                i = j;
                nums[++index] = nums[j];
            } else if (j - i < 2) {
                nums[++index] = nums[j];
            }
        }
        return index + 1;
    }

    public int removeDuplicates1(int[] nums) {
        int nLen = nums.length;
        if (nLen <= 2) return nLen;
        int i = 2, j = 2;
        while (j < nLen) {
            if (nums[i - 2] != nums[j]) {
                nums[i++] = nums[j];
            }
            j++;
        }
        return i;
    }

    public static void main(String[] args) {
        int[] nums = new int[]{1,1,1,1, 1,1, 1,2,2,3};
        int len = new Solution80().removeDuplicates1(nums);
        for (int i = 0; i < len; i++) {
            System.out.print(nums[i]);
            System.out.print(", ");
        }
    }
}
